package com.alit.ibot.robot.wechat.util;

import java.util.HashMap;
import java.util.Map;

import com.alit.ibot.robot.wechat.data.MediaId;

/**
 * @author litong
 * @date 2019年2月13日_下午3:57:42 
 * @version 1.0 
 */
public class WechatMediaIdCache {
  private static Map<String, MediaId> mediaIdMap = new HashMap<>();

	/**
	 * 添加
	 * @param mediaId
	 */
	public static void add(MediaId mediaId) {
		mediaIdMap.put(mediaId.getSha1(), mediaId);
	}

	public static void add(String sha1, String url, String localfile, String mediaId, long expireTime) {
		mediaIdMap.put(sha1, new MediaId(sha1, url, localfile, mediaId, expireTime));
	}

	/**
	 * 获取
	 */
	public static MediaId get(String sha1) {
		MediaId mediaId = mediaIdMap.get(sha1);
		return mediaId;
	}

	/**
	 * 返回所有
	 */
	public static Map<String, MediaId> getAllMediaId() {
		return mediaIdMap;
	}
}
